package leetcode.editor.cn;

import java.util.*;

/**
 * @id: 15
 * @title: 三数之和
 */
 
//给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重
//复的三元组。 
//
// 注意：答案中不可以包含重复的三元组。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [-1,0,1,2,-1,-4]
//输出：[[-1,-1,2],[-1,0,1]]
// 
//
// 示例 2： 
//
// 
//输入：nums = []
//输出：[]
// 
//
// 示例 3： 
//
// 
//输入：nums = [0]
//输出：[]
// 
//
// 
//
// 提示： 
//
// 
// 0 <= nums.length <= 3000 
// -105 <= nums[i] <= 105 
// 
// Related Topics 数组 双指针 排序 
// 👍 4089 👎 0

public class P15ThreeSum {
    public static void main(String[] args) {
        Solution solution = new P15ThreeSum().new Solution();
        // todo 
        final List<List<Integer>> lists1 = solution.threeSum(new int[]{-1, 0, 1, 2, -1, -4});
//        final List<List<Integer>> lists1 = solution.threeSum(new int[]{3,-2,1,0});
        System.out.println(lists1);
    }
    
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        if (nums == null || nums.length < 3) {
            return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        // a
        for (int i = 0; i < nums.length; ++i) {
            if (i > 0 && nums[i-1] == nums[i]) {
                continue;
            }
            int a = nums[i];
            // b
            for (int j = i+1; j < nums.length; ++j) {
                if (j > i+1 && nums[j-1] == nums[j]) {
                    continue;
                }
                int right = nums.length-1;
                // c ,这里不使用双指针，使用单指针减低逻辑判断的复杂度
                // 如果a+b+c>0，那么说明c太大了，需要变小，所有往左移
                while (j < right && a + nums[j] + nums[right] > 0) {
                    right--;
                }
                // 如果b和c交会还没找到，说明不会再有了，退出
                if (j == right) {
                    break;
                }
                if (a + nums[j] + nums[right] == 0) {
                    List<Integer> l = new ArrayList<>(3);
                    l.add(a);
                    l.add(nums[j]);
                    l.add(nums[right]);
                    result.add(l);
                }
            }
        }
        return result;
    }
}
//leetcode submit region end(Prohibit modification and deletion)


}